package com.yupi.yuaiagent.advisor;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

public class LoveAppContextualQueryAugmenterFactory {

    /**
     * 上下文查询增强器
     * @return 上下文查询增强器
     */
    public static ContextualQueryAugmenter createInstance() {
        PromptTemplate emptyContextPromptTemplate = new PromptTemplate("""
                你应该输出下面的内容：
                抱歉，我只能回答恋爱相关的问题，别的没办法帮到您哦，
                有问题可以联系客服 QQ:1572284002
                """);
        return ContextualQueryAugmenter.builder()
                .allowEmptyContext(false)  // 如果关闭的话，会输出默认的提示语。  如果开启的话，不会改变用户提示词
                .emptyContextPromptTemplate(emptyContextPromptTemplate)
                .build();
    }
}
